package com.fg.pro.ai.utils;

import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel;
import jakarta.annotation.PostConstruct;
import lombok.Data;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.deepseek.DeepSeekChatModel;
import org.springframework.ai.ollama.OllamaChatModel;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashMap;

@Data
@Component
public class MoreAiOptions {

    @Autowired
    DashScopeChatModel dashScopeChatModel;

    @Autowired
    DeepSeekChatModel deepSeekChatModel;

    @Autowired
    OllamaChatModel ollamaChatModel;

    HashMap<String, ChatModel> models = new HashMap<>();

    @PostConstruct
    private void init(){
        models.put("dashScope",dashScopeChatModel);
        models.put("deepSeekChat",deepSeekChatModel);
        models.put("ollamaChat",ollamaChatModel);
    }

    private String platform;
    private String model;
    private Double temperature;
}
